import yaml
from model.yolo_util import train, load_classes

_, CLASSES = load_classes('./cfg/classes.names')

if __name__ == "__main__":
    with open('cfg/cfg.yaml', 'r') as loadfile:
        config = yaml.load_all(loadfile, Loader=yaml.FullLoader)
        config_all = [x for x in config]

    # train mode
    config = config_all[0]

    train(config, CLASSES)
